Skip to main content

Pierrotlc's group workspace

Balance D and G training - 32x32

What makes this group special?
Tags

distinctive-waterfall-159

Notes
Author
State
Failed
Start time
January 31st, 2022 1:28:08 AM
Runtime
1h 3m 20s
Tracked hours
1h 3m 16s
Run path
pierrotlc/AnimeStyleGAN/8h67rtaz
OS
Linux-5.15.15-76051515-generic-x86_64-with-glibc2.10
Python version
3.8.5
Git repository
git clone git@github.com:Futurne/AnimeStyleGAN.git
Git state
git checkout -b "distinctive-waterfall-159" f6bd0f9bd07f0aaf7374f3c044ea9506cd5e703a
Command
launch_training.py
System Hardware
CPU count16
GPU count1
GPU typeNVIDIA GeForce RTX 3080 Laptop GPU
W&B CLI Version
0.12.9
Config

Config parameters are your model's inputs. Learn more

  • {} 30 keys
    • 256
    • [] 2 items
      • 0.5
      • 0.99
    • [] 2 items
      • 0.5
      • 0.5
    • "<torch.utils.data.dataloader.DataLoader object at 0x7fb96753cf40>"
    • "cuda"
    • 32
    • 32
    • 0.3
    • 100
    • 0.8
    • 0.7
    • 0.0001
    • 0.0001
    • [] 3 items
      • 3
      • 8
      • 25
    • [] 3 items
      • 3
      • 8
      • 25
    • 128
    • 6
    • 1
    • 2
    • 4
    • "Discriminator( (first_conv): Sequential( (0): Dropout(p=0.3, inplace=False) (1): Conv2d(3, 6, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1)) ) (blocks): ModuleList( (0): DiscriminatorBlock( (convs): ModuleList( (0): Sequential( (0): Dropout(p=0.3, inplace=False) (1): Conv2d(6, 6, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False) (2): BatchNorm2d(6, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True) (3): LeakyReLU(negative_slope=0.01) ) (1): Sequential( (0): Dropout(p=0.3, inplace=False) (1): Conv2d(6, 6, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False) (2): BatchNorm2d(6, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True) (3): LeakyReLU(negative_slope=0.01) ) ) (downsample): Conv2d(6, 12, kernel_size=(4, 4), stride=(2, 2), padding=(1, 1)) ) (1): DiscriminatorBlock( (convs): ModuleList( (0): Sequential( (0): Dropout(p=0.3, inplace=False) (1): Conv2d(12, 12, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False) (2): BatchNorm2d(12, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True) (3): LeakyReLU(negative_slope=0.01) ) (1): Sequential( (0): Dropout(p=0.3, inplace=False) (1): Conv2d(12, 12, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False) (2): BatchNorm2d(12, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True) (3): LeakyReLU(negative_slope=0.01) ) ) (downsample): Conv2d(12, 24, kernel_size=(4, 4), stride=(2, 2), padding=(1, 1)) ) (2): DiscriminatorBlock( (convs): ModuleList( (0): Sequential( (0): Dropout(p=0.3, inplace=False) (1): Conv2d(24, 24, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False) (2): BatchNorm2d(24, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True) (3): LeakyReLU(negative_slope=0.01) ) (1): Sequential( (0): Dropout(p=0.3, inplace=False) (1): Conv2d(24, 24, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False) (2): BatchNorm2d(24, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True) (3): LeakyReLU(negative_slope=0.01) ) ) (downsample): Conv2d(24, 48, kernel_size=(4, 4), stride=(2, 2), padding=(1, 1)) ) (3): DiscriminatorBlock( (convs): ModuleList( (0): Sequential( (0): Dropout(p=0.3, inplace=False) (1): Conv2d(48, 48, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False) (2): BatchNorm2d(48, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True) (3): LeakyReLU(negative_slope=0.01) ) (1): Sequential( (0): Dropout(p=0.3, inplace=False) (1): Conv2d(48, 48, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False) (2): BatchNorm2d(48, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True) (3): LeakyReLU(negative_slope=0.01) ) ) (downsample): Conv2d(48, 96, kernel_size=(4, 4), stride=(2, 2), padding=(1, 1)) ) (4): DiscriminatorBlock( (convs): ModuleList( (0): Sequential( (0): Dropout(p=0.3, inplace=False) (1): Conv2d(96, 96, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False) (2): BatchNorm2d(96, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True) (3): LeakyReLU(negative_slope=0.01) ) (1): Sequential( (0): Dropout(p=0.3, inplace=False) (1): Conv2d(96, 96, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False) (2): BatchNorm2d(96, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True) (3): LeakyReLU(negative_slope=0.01) ) ) (downsample): Conv2d(96, 192, kernel_size=(4, 4), stride=(2, 2), padding=(1, 1)) ) ) (classify): Sequential( (0): Conv2d(192, 1, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False) (1): Flatten(start_dim=1, end_dim=-1) ) )"
    • "StyleGAN( (mapping): MappingNetwork( (norm): LayerNorm((32,), eps=1e-05, elementwise_affine=True) (layers): ModuleList( (0): Sequential( (0): Linear(in_features=32, out_features=32, bias=True) (1): LayerNorm((32,), eps=1e-05, elementwise_affine=True) (2): LeakyReLU(negative_slope=0.01) ) (1): Sequential( (0): Linear(in_features=32, out_features=32, bias=True) (1): LayerNorm((32,), eps=1e-05, elementwise_affine=True) (2): LeakyReLU(negative_slope=0.01) ) (2): Sequential( (0): Linear(in_features=32, out_features=32, bias=True) (1): LayerNorm((32,), eps=1e-05, elementwise_affine=True) (2): LeakyReLU(negative_slope=0.01) ) (3): Sequential( (0): Linear(in_features=32, out_features=32, bias=True) (1): LayerNorm((32,), eps=1e-05, elementwise_affine=True) (2): LeakyReLU(negative_slope=0.01) ) ) (out): Linear(in_features=32, out_features=32, bias=True) ) (synthesis): SynthesisNetwork( (blocks): ModuleList( (0): SynthesisBlock( (conv2): Sequential( (0): Dropout(p=0.3, inplace=False) (1): Conv2d(128, 128, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1)) (2): LeakyReLU(negative_slope=0.01) ) (ada_in): AdaIN() ) (1): SynthesisBlock( (upsample): ConvTranspose2d(128, 64, kernel_size=(4, 4), stride=(2, 2), padding=(1, 1)) (conv1): Sequential( (0): Dropout(p=0.3, inplace=False) (1): Conv2d(64, 64, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1)) (2): LeakyReLU(negative_slope=0.01) ) (conv2): Sequential( (0): Dropout(p=0.3, inplace=False) (1): Conv2d(64, 64, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1)) (2): LeakyReLU(negative_slope=0.01) ) (ada_in): AdaIN() ) (2): SynthesisBlock( (upsample): ConvTranspose2d(64, 32, kernel_size=(4, 4), stride=(2, 2), padding=(1, 1)) (conv1): Sequential( (0): Dropout(p=0.3, inplace=False) (1): Conv2d(32, 32, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1)) (2): LeakyReLU(negative_slope=0.01) ) (conv2): Sequential( (0): Dropout(p=0.3, inplace=False) (1): Conv2d(32, 32, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1)) (2): LeakyReLU(negative_slope=0.01) ) (ada_in): AdaIN() ) (3): SynthesisBlock( (upsample): ConvTranspose2d(32, 16, kernel_size=(4, 4), stride=(2, 2), padding=(1, 1)) (conv1): Sequential( (0): Dropout(p=0.3, inplace=False) (1): Conv2d(16, 16, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1)) (2): LeakyReLU(negative_slope=0.01) ) (conv2): Sequential( (0): Dropout(p=0.3, inplace=False) (1): Conv2d(16, 16, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1)) (2): LeakyReLU(negative_slope=0.01) ) (ada_in): AdaIN() ) (4): SynthesisBlock( (upsample): ConvTranspose2d(16, 8, kernel_size=(4, 4), stride=(2, 2), padding=(1, 1)) (conv1): Sequential( (0): Dropout(p=0.3, inplace=False) (1): Conv2d(8, 8, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1)) (2): LeakyReLU(negative_slope=0.01) ) (conv2): Sequential( (0): Dropout(p=0.3, inplace=False) (1): Conv2d(8, 8, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1)) (2): LeakyReLU(negative_slope=0.01) ) (ada_in): AdaIN() ) ) (to_rgb): Conv2d(8, 3, kernel_size=(1, 1), stride=(1, 1)) ) )"
    • "Adam ( Parameter Group 0 amsgrad: False betas: (0.5, 0.99) eps: 1e-08 initial_lr: 0.0001 lr: 0.0001 weight_decay: 0 )"
    • "Adam ( Parameter Group 0 amsgrad: False betas: (0.5, 0.5) eps: 1e-08 initial_lr: 0.0001 lr: 0.0001 weight_decay: 0 )"
    • 0
    • "<torch.optim.lr_scheduler.MultiStepLR object at 0x7fb9715e9310>"
    • "<torch.optim.lr_scheduler.MultiStepLR object at 0x7fb96753cc40>"
    • 0
    • 0
    • 1
Summary

Summary metrics are your model's outputs. Learn more

  • {} 7 keys
    • 0.4672722767395188
    • 0.7990070715486763
    • 1.1307418686319068
    • 0.6764448467028668
    • 1.135438687351334
    • {} 7 keys
      • 0.29685521616514426
    Artifact Outputs

    This run produced these artifacts as outputs. Total: 1. Learn more

    Loading...